import torch

from src.nets.MRNet import MRNet

model_path = './logs/ep003-loss1.670690.pth'
model = MRNet()
model.load_state_dict(torch.load(model_path))
print(model)

